python -m torch.distributed.launch --nproc_per_node=8 --master_port 18208 --use_env main_imagenet.py --model regnet_x_1_6gf --epochs 100 --batch-size 256 --wd 0.00005 --lr 0.08 --lr-scheduler=cosineannealinglr --lr-warmup-method=linear --lr-warmup-epochs=5 --lr-warmup-decay=0.1 --prune --cache-dataset --method group_norm --soft-keeping-ratio 0.6 --pretrained --output-dir run/imagenet/regnet_x_1_6gf_gnorm --target-flops 0.8 --global-pruning --workers 16 --print-freq 100 --amp